import os
import pytest
from selenium import webdriver
from selenium.webdriver.chrome.service import Service

# 指定 chromedriver 的完整路径
CHROMEDRIVER_PATH = os.path.join("D:\\", "python39", "Scripts", "chromedriver.exe")

@pytest.fixture()
def setup_browser(request):
    """
    浏览器设置夹具（fixture）。
    """
    # 使用指定路径的 chromedriver
    driver = webdriver.Chrome(service=Service(CHROMEDRIVER_PATH))

    # 设置浏览器窗口大小
    driver.maximize_window()

    # 定义一个 teardown 函数，用于在测试结束时关闭浏览器
    def teardown():
        driver.quit()

    # 将 teardown 函数注册为 finalizer
    request.addfinalizer(teardown)

    return driver
